KYBER_VERSION = None
KYBER_N = None
KYBER_k = None
ETA1 = None
ETA2 = None
KYBER_d = None

parameter_set = {512: [256, 2, 3, 2, (10, 4)], 768: [256, 3, 2, 2, (10, 4)], 1024: [256, 4, 2, 2, (11, 5)]}


def set_version(ver):
    global KYBER_VERSION, KYBER_N, KYBER_k, ETA1, ETA2, KYBER_d
    KYBER_VERSION = ver

    KYBER_N, KYBER_k, ETA1, ETA2, KYBER_d = parameter_set[int(KYBER_VERSION)]
